import os

# 指定包含文件的文件夹路径
folder_path = "../images/train"

# 遍历文件夹中的所有文件
for filename in os.listdir(folder_path):
    file_name, file_extension = os.path.splitext(filename)
    # 检查文件名是否符合指定格式，这里假设文件名以"2022_"开头并以".txt"结尾
    if file_name.endswith("_1"):
        os.remove(os.path.join(folder_path, filename))
    elif file_name.endswith("_0"):
        # # 提取出"_"后面的内容
        # new_filename = filename.split("_", 1)[1].lstrip("0")  # 使用split函数以_为分隔符，保留第二个部分，然后使用lstrip函数移除开头的0
        file_name = file_name[:-2]
        new_filename = file_name + file_extension
        # 构造新的文件路径
        new_filepath = os.path.join(folder_path, new_filename)
        # 重命名文件
        os.rename(os.path.join(folder_path, filename), new_filepath)
